/*================================================================
*   文件名称：main.cpp
*   创 建 者：yang qiang
*   创建日期：2019年02月24日
*   描    述：
*   
================================================================*/


#include <iostream>
#include <vector>
using namespace std;

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        if(m == 0)
            return;
        int n = matrix[0].size();
        if(n == 0)
            return;

        int flag = 0;
        for(int i = 0; i < m; i++){
            if(matrix[i][0] == 0)
                flag = 1;
            for(int j = 1; j < n; j++){
                if(matrix[i][j] == 0){
                    matrix[0][j] = matrix[i][0] = 0;
                }
            }
        }

        for(int i = m - 1; i >= 0; i--){
            for( int j = n - 1; j >= 1; j-- ){
                if(matrix[i][0] == 0 || matrix[0][j] == 0)
                    matrix[i][j] = 0;
            }
        }
        
        if( flag == 1 )
            for(int i = 0; i < m; i++){
                matrix[i][0] = 0;
            }

        return;
    }
};
